Fix up unit tests for disk usage collector to be correct now