fixed my tests