put test back how it should be