Fixed bad class ref.