diff --git a/django_rest_passwordreset/migrations/0002_pk_migration.py b/django_rest_passwordreset/migrations/0002_pk_migration.py index 27c830b..c846b4f 100644 --- a/django_rest_passwordreset/migrations/0002_pk_migration.py +++ b/django_rest_passwordreset/migrations/0002_pk_migration.py @@ -9,9 +9,9 @@ def populate_auto_incrementing_pk_field(apps, schema_editor): ResetPasswordToken = apps.get_model('django_rest_passwordreset', 'ResetPasswordToken') - + db_alias = schema_editor.connection.alias # Generate values for the new id column - for i, o in enumerate(ResetPasswordToken.objects.all()): + for i, o in enumerate(ResetPasswordToken.objects.using(db_alias).all()): o.id = i + 1 o.save()