[FEATURE] Small changes to the disk usage test.