diff --git a/main.py b/main.py
index 9910f22..0012537 100644
--- a/main.py
+++ b/main.py
@@ -145,7 +145,18 @@ class Berry(App):
 				else:
 					code_editor.text = f.read()
 				
-				code_editor.language = theme_mappings.get(self.open_file.rsplit(".", 1)[1], None) 
+				file_extension = self.open_file
+				dot_count = file_extension.count(".")
+				
+				if dot_count == 1:
+					if file_extension.startswith("."):
+						file_extension = file_extension.removeprefix(".")
+					else:
+						file_extension = file_extension.rsplit(".", 1)[1]
+				elif dot_count > 1:
+					file_extension = file_extension.rsplit(".", 1)[1]
+
+				code_editor.language = theme_mappings.get(file_extension, None) 
 				code_editor.disabled = False
 			except UnicodeDecodeError:
 				code_editor.text = "This file is in binary, it can't be openned. Sorry."