For ENG-5959, fix test bug