trying to improve tests