controller classes fix