Fix bug with incorrect class name