Fix problem with missing migration