make tests better