This actually seems wrong also - if the storage level is useMemory + useDisk, wouldn't this only include the entry for diskSize in the hash map?