Restart failed units - fixes 326

Fixes #326.