Cleanup and fixery.

1 files changed, 4 insertions, 2 deletions

diff --git a/src/uk/org/ury/library/viewer/LibraryViewerPanel.java b/src/uk/org/ury/library/viewer/LibraryViewerPanel.java
index e05164e..b3ffdd3 100644
--- a/src/uk/org/ury/library/viewer/LibraryViewerPanel.java
+++ b/src/uk/org/ury/library/viewer/LibraryViewerPanel.java
@@ -15,13 +15,15 @@ import javax.swing.JTable;
 import javax.swing.JTextField;
 import javax.swing.SwingWorker;
 
-import uk.org.ury.database.exceptions.QueryFailureException;
 import uk.org.ury.frontend.FrontendMaster;
 import uk.org.ury.frontend.FrontendModulePanel;
 import uk.org.ury.frontend.HintField;
 import uk.org.ury.frontend.exceptions.UICreationFailureException;
+
 import uk.org.ury.library.exceptions.EmptySearchException;
 
+import uk.org.ury.protocol.exceptions.InvalidMessageException;
+
 
 /**
  * Frontend panel providing access to an underlying library viewer.
@@ -173,7 +175,7 @@ public class LibraryViewerPanel extends FrontendModulePanel
           {
             master.doSearch (searchTerm);
           }
-        catch (QueryFailureException e)
+        catch (InvalidMessageException e)
           {
             searchFailureMessage = String.format (rb.getString ("ERR_SEARCH_FAILED"),
                                                   searchTerm, e.getMessage ());