fixed incorrect class assignment