oops, changed the class