Fix code formatting in AddStateToContext migration