bug fix

replaced bad class-name code