FIX: Make testing slightly cleaner

This does not fix the problems in `master`, but it should make them easier to diagnose. Ready for review/merge from my end.