Fixed mistake with incorrect class name.