Update class button fix