diff --git a/db_wrapper.py b/db_wrapper.py
index ff3b75c..0d63017 100644
--- a/db_wrapper.py
+++ b/db_wrapper.py
@@ -1,5 +1,6 @@
 import logging
 import psycopg2
+import psycopg2.extras
 import sqlite3