diff --git a/README.md b/README.md
index 24d9bb9..ce43bbe 100644
--- a/README.md
+++ b/README.md
@@ -51,6 +51,7 @@ Or check out some examples:
 Create a migration to enable the extension
 
 ```python
+from django.db import migrations
 from pgvector.django import VectorExtension
 
 class Migration(migrations.Migration):