diff --git a/mypy-baseline.txt b/mypy-baseline.txt
index 3c6bbf22089c0..9b607b6222cd3 100644
--- a/mypy-baseline.txt
+++ b/mypy-baseline.txt
@@ -366,6 +366,7 @@ posthog/session_recordings/queries/session_recording_list_from_replay_summary.py
 posthog/session_recordings/queries/session_recording_list_from_replay_summary.py:0: note: If the method is meant to be abstract, use @abc.abstractmethod
 posthog/session_recordings/queries/session_recording_list_from_replay_summary.py:0: error: Missing return statement  [empty-body]
 posthog/session_recordings/queries/session_recording_list_from_replay_summary.py:0: note: If the method is meant to be abstract, use @abc.abstractmethod
+posthog/migrations/0404_remove_propertydefinition_property_type_is_valid_and_more.py:0: error: Module "django.contrib.postgres.operations" has no attribute "AddConstraintNotValid"  [attr-defined]
 posthog/hogql_queries/test/test_query_runner.py:0: error: Variable "TestQueryRunner" is not valid as a type  [valid-type]
 posthog/hogql_queries/test/test_query_runner.py:0: note: See https://mypy.readthedocs.io/en/stable/common_issues.html#variables-vs-type-aliases
 posthog/hogql_queries/test/test_query_runner.py:0: error: Invalid base class "TestQueryRunner"  [misc]
diff --git a/posthog/migrations/0404_remove_propertydefinition_property_type_is_valid_and_more.py b/posthog/migrations/0404_remove_propertydefinition_property_type_is_valid_and_more.py
index ae84002a79b58..a90c8214064bd 100644
--- a/posthog/migrations/0404_remove_propertydefinition_property_type_is_valid_and_more.py
+++ b/posthog/migrations/0404_remove_propertydefinition_property_type_is_valid_and_more.py
@@ -1,5 +1,5 @@
 # Generated by Django 4.2.11 on 2024-04-21 21:11
-
+from django.contrib.postgres.operations import AddConstraintNotValid
 from django.db import migrations, models
 
 
@@ -36,7 +36,7 @@ class Migration(migrations.Migration):
                 choices=[(1, "event"), (2, "person"), (3, "group"), (4, "session")], default=1
             ),
         ),
-        migrations.AddConstraint(
+        AddConstraintNotValid(
             model_name="propertydefinition",
             constraint=models.CheckConstraint(
                 check=models.Q(("property_type__in", ["DateTime", "String", "Numeric", "Boolean", "Duration"])),