fix tests #2