Refactor SampleVNF wait method