Tried to fix test