diff --git a/babel.cfg b/babel.cfg
index cbdd8fa5..dcdd8e34 100644
--- a/babel.cfg
+++ b/babel.cfg
@@ -1,6 +1,8 @@
 [python: src/octoprint/**.py]
 [jinja2: src/octoprint/templates/**.jinja2]
+[jinja2: src/octoprint/plugins/**.jinja2]
 extensions=jinja2.ext.autoescape, jinja2.ext.with_
 
 [javascript: src/octoprint/static/js/app/**.js]
+[javascript: src/octoprint/plugins/**.js]
 extract_messages = gettext, ngettext
diff --git a/setup.py b/setup.py
index 78df56c7..17549a0e 100644
--- a/setup.py
+++ b/setup.py
@@ -194,7 +194,7 @@ class RefreshTranslation(Command):
 		self.babel_extract_messages.copyright_holder = "The OctoPrint Project"
 		self.babel_extract_messages.finalize_options()
 
-		self.babel_update_messages.input_file = I18N_MAPPING_FILE
+		self.babel_update_messages.input_file = I18N_POT_FILE
 		self.babel_update_messages.output_dir = I18N_OUTPUT_DIR_PY
 		self.babel_update_messages.locale = self.locale