Fixed class error