Fix for new github layout class names

Updated class name.