made tests better