diff --git a/strategy/strategy_db.py b/strategy/strategy_db.py
index 5432f5d..d7792ca 100644
--- a/strategy/strategy_db.py
+++ b/strategy/strategy_db.py
@@ -6,6 +6,8 @@
 """策略相关表结构。"""
 from __future__ import annotations
 
+from db_conn import rollback_if_postgres
+
 ROLL_GROUPS_SQL = """
 CREATE TABLE IF NOT EXISTS roll_groups (
     id INTEGER PRIMARY KEY AUTOINCREMENT,