Attempt to fix test