diff --git a/api/s34_sa_gen.py b/api/s34_sa_gen.py
index 8aaaa00..7eb7ff8 100644
--- a/api/s34_sa_gen.py
+++ b/api/s34_sa_gen.py
@@ -2,6 +2,7 @@ from .s32_region_util import calculate_boundary, inflate_boundary
 from .s34_sa_cal import *
 from .s34_sa import get_all_service_area_ids
 from .batch_exe import execute_batch_command
+from .database import ChangeSet
 
 def generate_service_area(name: str, inflate_delta: float = 0.5) -> ChangeSet:
     cs = ChangeSet()