diff --git a/.gitignore b/.gitignore
index b6b50a2f..2c4abea7 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,7 +2,7 @@
 *map
 
 # nix
-result
+/result
 
 # extras
 backport/
diff --git a/flake.nix b/flake.nix
index 5721e9b1..31ff8e3e 100644
--- a/flake.nix
+++ b/flake.nix
@@ -12,7 +12,7 @@
     flake-utils.lib.eachDefaultSystem (
       system: let
         name = "smassh";
-        version = "3.1.5";
+        version = "3.1.6";
 
         pkgs = import nixpkgs {inherit system;};
         python3 = pkgs.python312Packages;
diff --git a/pyproject.toml b/pyproject.toml
index 5604056f..419b4ab7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "smassh"
-version = "3.1.5"
+version = "3.1.6"
 description = "A TUI based typing application"
 maintainers = ["kraanzu <kraanzu@gmail.com>"]
 license = "GPL v3"
@@ -33,4 +33,4 @@ smassh = 'smassh.__main__:main'
 
 [tool.ruff]
 exclude = ["venv"]
-ignore = ["F405"]
+lint.ignore = ["F405"]
diff --git a/smassh/__init__.py b/smassh/__init__.py
index 1eeed0f1..259976ca 100644
--- a/smassh/__init__.py
+++ b/smassh/__init__.py
@@ -1,6 +1,6 @@
 import click
 
-PKG_VERSION = "3.1.4"
+PKG_VERSION = "3.1.6"
 
 
 @click.group(
diff --git a/smassh/__main__.py b/smassh/__main__.py
index 1eeed0f1..259976ca 100644
--- a/smassh/__main__.py
+++ b/smassh/__main__.py
@@ -1,6 +1,6 @@
 import click
 
-PKG_VERSION = "3.1.4"
+PKG_VERSION = "3.1.6"
 
 
 @click.group(
diff --git a/smassh/ui/tui.py b/smassh/ui/tui.py
index 1deb0021..f5d6cf33 100644
--- a/smassh/ui/tui.py
+++ b/smassh/ui/tui.py
@@ -101,14 +101,14 @@ async def on_mount(self) -> None:
     @on(ApplyLanguage)
     def apply_language(self, event: ApplyLanguage) -> None:
         config_parser.set("language", event.value)
-        self.SCREENS["main"].query_one(LanguagePalette).refresh()
-        self.SCREENS["main"].query_one(Space).reset()
+        self.app.get_screen("main").query_one(LanguagePalette).refresh()
+        self.app.get_screen("main").query_one(Space).reset()
 
     @on(ApplyTheme)
     def apply_theme(self, event: ApplyTheme) -> None:
         self.action_theme(event.value)
         config_parser.set("theme", event.value)
-        self.SCREENS["main"].query_one(ThemePalette).refresh()
+        self.app.get_screen("main").query_one(ThemePalette).refresh()
 
     def action_star(self) -> None:
         webbrowser.open("https://github.com/kraanzu/smassh")
diff --git a/smassh/ui/widgets/palette/palette_list.py b/smassh/ui/widgets/palette/palette_list.py
index 1c7f2ca9..efcca5e6 100644
--- a/smassh/ui/widgets/palette/palette_list.py
+++ b/smassh/ui/widgets/palette/palette_list.py
@@ -66,7 +66,7 @@ def valid_option(option_text: str) -> bool:
         valid_options = filter(valid_option, self.get_options())
         self.add_options(valid_options)
 
-    async def on_mount(self, _) -> None:
+    def on_mount(self) -> None:
         self.apply_filter("")
         options = sorted(self._get_options())
         index = options.index(self.get_current())