From 9d1733df07296bcdf9c196cec5f470bc66b92b2f Mon Sep 17 00:00:00 2001
From: Aidar Samerkhanov <aidar.samerkhanov@memgraph.io>
Date: Tue, 16 May 2023 09:15:20 +0000
Subject: [PATCH] Fix Get/SetStorageMode for DiskStorage.

---
 src/storage/v2/disk/storage.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/storage/v2/disk/storage.cpp b/src/storage/v2/disk/storage.cpp
index ad48a8571..85bc3e765 100644
--- a/src/storage/v2/disk/storage.cpp
+++ b/src/storage/v2/disk/storage.cpp
@@ -1762,12 +1762,12 @@ bool DiskStorage::AppendToWalDataDefinition(durability::StorageGlobalOperation o
   throw utils::NotYetImplemented("AppendToWalDataDefinition");
 }
 
-void InMemoryStorage::SetStorageMode(StorageMode storage_mode) {
+void DiskStorage::SetStorageMode(StorageMode storage_mode) {
   std::unique_lock main_guard{main_lock_};
   storage_mode_ = storage_mode;
 }
 
-StorageMode InMemoryStorage::GetStorageMode() { return storage_mode_; }
+StorageMode DiskStorage::GetStorageMode() { return storage_mode_; }
 
 utils::BasicResult<DiskStorage::CreateSnapshotError> DiskStorage::CreateSnapshot(std::optional<bool> is_periodic) {
   throw utils::NotYetImplemented("CreateSnapshot");