Fixed error with class assignment.