Use storage module in tests