class error fixed