Fixed bug with class assignment.