Fix reference to Storage class in tests