diff --git a/README.md b/README.md
index b239025..54a04d8 100644
--- a/README.md
+++ b/README.md
@@ -47,7 +47,7 @@ PyPi is not used for CaskDB yet ([issue #5](https://github.com/avinassh/py-caskd
 ## Usage
 
 ```python
-disk: DiskStorage = DiskStore(file_name="books.db")
+disk: DiskStorage = DiskStorage(file_name="books.db")
 disk.set(key="othello", value="shakespeare")
 author: str = disk.get("othello")
 # it also supports dictionary style API too:
diff --git a/src/caskdb/disk_store.py b/src/caskdb/disk_store.py
index 89f18a7..140fad2 100644
--- a/src/caskdb/disk_store.py
+++ b/src/caskdb/disk_store.py
@@ -13,7 +13,7 @@
 
 Typical usage example:
 
-    disk: DiskStorage = DiskStore(file_name="books.db")
+    disk: DiskStorage = DiskStorage(file_name="books.db")
     disk.set(key="othello", value="shakespeare")
     author: str = disk.get("othello")
     # it also supports dictionary style API too: