diff --git a/app/tests/test_api.py b/app/tests/test_api.py
index 154e178..98d2352 100644
--- a/app/tests/test_api.py
+++ b/app/tests/test_api.py
@@ -1,7 +1,7 @@
 import pytest
 from app import app
 from app.default_data import populate_test_data
-from app.models import db, License, Tag, User, UserRank, Package
+from app.models import db, License, Tag, User, UserRank, Package, PackageState
 from utils import client, recreate_db, parse_json
 from utils import is_str, is_int, is_optional