Adapt storage benchmark to use shallow storage