fix tests

#29