diff --git a/src/zhinst/toolkit/driver/devices/__init__.py b/src/zhinst/toolkit/driver/devices/__init__.py
index 4ae50e85..c20ca206 100644
--- a/src/zhinst/toolkit/driver/devices/__init__.py
+++ b/src/zhinst/toolkit/driver/devices/__init__.py
@@ -24,6 +24,7 @@
     "PQSC": PQSC,
     "UHFQA": UHFQA,
     "UHFLI": UHFLI,
+    "UHFAWG": UHFLI,
 }
 
 __all__ = [