diff --git a/zenmap/zenmapGUI/ScanHostsView.py b/zenmap/zenmapGUI/ScanHostsView.py
index 38f924650..d7af137e3 100644
--- a/zenmap/zenmapGUI/ScanHostsView.py
+++ b/zenmap/zenmapGUI/ScanHostsView.py
@@ -345,7 +345,7 @@ class ScanHostsView(HIGVBox, object):
 
 if __name__ == "__main__":
     w = gtk.Window()
-    h = ScanHostsView()
+    h = ScanHostsView(None)
     w.add(h)
     w.show_all()
     gtk.main()
diff --git a/zenmap/zenmapGUI/SearchWindow.py b/zenmap/zenmapGUI/SearchWindow.py
index aed11c537..6c38729ec 100644
--- a/zenmap/zenmapGUI/SearchWindow.py
+++ b/zenmap/zenmapGUI/SearchWindow.py
@@ -247,6 +247,6 @@ class SearchWindow(BaseSearchWindow, object):
 
 
 if __name__ == "__main__":
-    search = SearchWindow(lambda x: gtk.main_quit())
+    search = SearchWindow(lambda x: gtk.main_quit(), lambda x: gtk.main_quit())
     search.show_all()
     gtk.main()