diff --git a/cleanmedia b/cleanmedia
index 37e0ff6..18486d2 100755
--- a/cleanmedia
+++ b/cleanmedia
@@ -23,7 +23,7 @@ from pathlib import Path
 from typing import Optional, Union, List, Tuple
 
 try:
-    import psycopg2
+    import psycopg2, psycopg2.extensions
     import yaml
 except ImportError:
     raise Exception("Please install psycopg2 and pyyaml")