diff --git a/haxor_news/haxor.py b/haxor_news/haxor.py
index ebb813c..ebff69e 100644
--- a/haxor_news/haxor.py
+++ b/haxor_news/haxor.py
@@ -152,7 +152,7 @@ def _create_cli(self):
             enable_open_in_editor=True,
             enable_system_prompt=True,
             enable_suspend=True,
-            extra_key_bindings=self.key_manager.key_bindings)
+            key_bindings=self.key_manager.key_bindings)
 
     def _add_comment_pagination(self, document_text):
         """Add the command to enable comment pagination where applicable.
diff --git a/haxor_news/style.py b/haxor_news/style.py
index 3470840..49a7d46 100644
--- a/haxor_news/style.py
+++ b/haxor_news/style.py
@@ -15,7 +15,7 @@
 
 from pygments.token import Token
 from pygments.util import ClassNotFound
-from prompt_toolkit.styles.pygments import style_from_pygments_dict, style_from_pygments
+from prompt_toolkit.styles.pygments import style_from_pygments_dict, style_from_pygments_cls
 from prompt_toolkit.styles import merge_styles
 import pygments.styles
 
@@ -48,7 +48,7 @@ def style_factory(self, name):
 
         # Create styles dictionary.
         return merge_styles([
-            style_from_pygments(pygments_style),
+            style_from_pygments_cls(pygments_style),
             style_from_pygments_dict({
                 Token.Menu.Completions.Completion.Current: 'bg:#00aaaa #000000',
                 Token.Menu.Completions.Completion: 'bg:#008888 #ffffff',