fix for class