small fix tests