For ENG-5048, fix test bug