Edit to migration to allow it on staging