fix tests and add update class