diff --git a/src/wiktextract_wrapper.py b/src/wiktextract_wrapper.py
index 2d6bc48602c16f32fe838e73974608b23d7d6905..47e68417b47f46e7afa4ea956e91143ceb84660b 100755
--- a/src/wiktextract_wrapper.py
+++ b/src/wiktextract_wrapper.py
@@ -14,17 +14,18 @@ if __name__ == "__main__":
 
     args = parser.parse_args()
     formerdir = None
-    if args.zero_config:
-        import sys
-        import pathlib
-        #print(f"zero config {pathlib.Path(__file__).parent.parent.resolve()}")
-        formerdir = os.getcwd()
-        os.chdir(pathlib.Path(__file__).parent.parent.resolve())
+
+if ('args' in dir() and args.zero_config) or 'zero_config' in dir():
+    import sys
+    import pathlib
+    #print(f"zero config {pathlib.Path(__file__).parent.parent.resolve()}")
+    formerdir = os.getcwd()
+    os.chdir(pathlib.Path(__file__).parent.parent.resolve())
 
 import time
 t = time.time()
 from wiktextract.page import parse_page
-if args.show_timings:
+if 'args' in dir() and args.show_timings:
     print(time.time()-t)
 from wiktextract_context import get_wiktextract_context
 import json
@@ -69,7 +70,7 @@ class Wiktextract:
             transformed_pos = {
                 "sources": [
                     {
-                        "wiktionary_language": pos.get("lang_code", ""),  # Assuming this key is present in your data
+                        "wiktionary_language": self.wiktlang,  # Assuming this key is present in your data
                         "wiktextract_version": metadata.version("wiktextract"),
                         "wrapper_version": git.Repo(search_parent_directories=True).head.object.hexsha
                     }