trying to fix tests but not working for some reason