fix bug of spark "SubmitStage" listener as unit test error