Fix prefect restart unit testing

[huan]

for (let n = 0; n < 3; n++) {
  await puppet.start()
  await puppet.stop()
}

Fix

• 155

[CLAassistant]

All committers have signed the CLA.