really fix tests