Restart failed units - fixes 326

Fixes #326.
If this is going to return a set, it might be better to not create `incomplete_units` as a list and then cast it to a set, but instead create a set in the first place.