Fixed rolling back migrations

Without this, rolling back migrations in the cli is broken.