Fixed faulty class name getting in warning code