diff --git a/tests/test_databases.py b/tests/test_databases.py
index 8bffe33b4..eb7500ea8 100644
--- a/tests/test_databases.py
+++ b/tests/test_databases.py
@@ -1,6 +1,7 @@
 import ase
 
 import aiidalab_widgets_base as awb
+from aiidalab_widgets_base import ComputationalResourcesDatabaseWidget
 
 
 def test_cod_query_widget():