Fix for showing correct student class in update