Adds missing model information to to custom migration