class fixed
Working