Fixed small mistake in migration