trying to fix these tests