Fixed mismatched class