diff --git a/pan_os_upgrade/upgrade.py b/pan_os_upgrade/upgrade.py
index 0d72d25..3ded58d 100644
--- a/pan_os_upgrade/upgrade.py
+++ b/pan_os_upgrade/upgrade.py
@@ -32,7 +32,7 @@
 from pydantic import BaseModel
 
 # project imports
-from models import SnapshotReport, ReadinessCheckReport
+from pan_os_upgrade.models import SnapshotReport, ReadinessCheckReport
 
 
 # ----------------------------------------------------------------------------