Add migration to fix column type