Trying to fix test