Add migrations back in