made tests work except for one