diff --git a/tests/code-gen/generator.py b/tests/code-gen/generator.py
index f4af06e..55a393c 100755
--- a/tests/code-gen/generator.py
+++ b/tests/code-gen/generator.py
@@ -39,7 +39,8 @@
       pass
 
 script_dir = os.path.dirname(os.path.realpath(__file__))
-db_file_path = os.path.join(script_dir, '/../../tensorforge/arch_db.yml')
+print('script_dir: ', script_dir)
+db_file_path = os.path.join('./tensorforge/arch_db.yml')
 with open(db_file_path, 'r') as file:
   yaml_data = yaml.safe_load(file)
 cpu_archs = [item['arch'] for item in yaml_data]