[Fix] Fixed failing test in previous pull request.