fix tests part one