diff --git a/docs/symlink_py.bash b/docs/symlink_py.bash
index d8ebe578f..df8cf37f7 100755
--- a/docs/symlink_py.bash
+++ b/docs/symlink_py.bash
@@ -6,14 +6,21 @@ cd links_to_compiler_build/py/concretelang_core
 
 ln -s ../../../../compiler/build/tools/concretelang/python_packages/concretelang_core/mlir -f
 
-mkdir -p concrete/lang
-
+# Create directories needed for symlinks
+mkdir -p concrete/lang/dialects
 cd concrete
+# Consider concrete as a package, as it's not detecting it as a namespace
+touch __init__.py
 
-ln -s ../../../../../compiler/build/tools/concretelang/python_packages/concretelang_core/concrete/lang/dialects/_FHE_ops_gen.py lang/fhe.py  -f
-ln -s ../../../../../compiler/build/tools/concretelang/python_packages/concretelang_core/concrete/lang/dialects/_FHELinalg_ops_gen.py lang/fhelinalg.py  -f
-ln -s ../../../../../compiler/build/tools/concretelang/python_packages/concretelang_core/concrete/lang/dialects/_ods_common.py lang/_ods_common.py  -f
+py_prefix="$PWD/../../../../../compiler/build/tools/concretelang/python_packages/concretelang_core/concrete/"
+pyfiles=`find $py_prefix -iname "*.py"`
 
-ln -s ../../../../../compiler/build/tools/concretelang/python_packages/concretelang_core/concrete/compiler.py compiler.py  -f
+for file in $pyfiles
+do
+    ln -s $file `echo $file | sed s:$py_prefix::` -f
+done
 
-touch lang/__init__.py __init__.py
+# Manually create dialect files
+ln -s ${py_prefix}lang/dialects/_FHE_ops_gen.py lang/dialects/fhe.py  -f
+ln -s ${py_prefix}lang/dialects/_FHELinalg_ops_gen.py lang/dialects/fhelinalg.py  -f
+ln -s ${py_prefix}lang/dialects/_ods_common.py lang/dialects/_ods_common.py  -f