fix tests units