Mini cleanup in tests