diff --git a/gds_write/def2stream.py b/gds_write/def2stream.py
index 0308d428..a0a1eea1 100755
--- a/gds_write/def2stream.py
+++ b/gds_write/def2stream.py
@@ -38,7 +38,7 @@
     * from https://github.com/The-OpenROAD-Project/OpenROAD-flow-scripts/blob/99e8d3e9b399c7703fab26bef2b5770d39a2f724/flow/util/def2stream.py
 """
 
-import klayout.db as db
+from klayout import db
 import re
 import argparse