diff --git a/src/gnomeshellsearch.py b/src/gnomeshellsearch.py
index 8aa24a2..63479f4 100644
--- a/src/gnomeshellsearch.py
+++ b/src/gnomeshellsearch.py
@@ -1,7 +1,6 @@
 # -*- coding: utf-8 -*-
 # Copyright 2016-2017 Andrew Chadwick <a.t.chadwick@gmail.com>
 # Copyright 2014-2015 Davi da Silva Böger <dsboger@gmail.com>
-
 """Zim plugin to display Zim pages in GNOME Shell search results."""
 
 import json
@@ -19,7 +18,6 @@
 logger = logging.getLogger(__name__)
 ZIM_COMMAND = "zim"
 
-
 # Zim plugin inegration:
 
 class GnomeShellSearchPluginCommand (NotebookCommand):
@@ -95,14 +93,14 @@ def __init__(self, notebook=None, search_all=True):
         self.search_all = search_all
 
     def main(self):
-        import gtk
+        from gi.repository import Gtk
 
-        gtk.main()
+        Gtk.main()
 
     def quit(self):
-        import gtk
+        from gi.repository import Gtk
 
-        gtk.main_quit()
+        Gtk.main_quit()
 
     @dbus.service.method(dbus_interface=SEARCH_IFACE,
                          in_signature='as', out_signature='as',