Fixes class name