Adding fix for migrations