Fixes issue with missing migration