diff --git a/swh/storage/__init__.py b/swh/storage/__init__.py
index 63ecdaaf1..4c306f967 100644
--- a/swh/storage/__init__.py
+++ b/swh/storage/__init__.py
@@ -1,40 +1,43 @@
 # Copyright (C) 2015-2016  The Software Heritage developers
 # See the AUTHORS file at the top-level directory of this distribution
 # License: GNU General Public License version 3, or any later version
 # See top-level LICENSE file for more information
 
 from . import storage
 
 Storage = storage.Storage
 
 
 class HashCollision(Exception):
     pass
 
 
 def get_storage(cls, args):
     """
     Get a storage object of class `storage_class` with arguments
     `storage_args`.
 
     Args:
         storage (dict): dictionary with keys:
-        - cls (str): storage's class, either 'local' or 'remote'
+        - cls (str): storage's class, either 'local', 'remote',
+                     or 'memory'
         - args (dict): dictionary with keys
 
     Returns:
         an instance of swh.storage.Storage (either local or remote)
 
     Raises:
         ValueError if passed an unknown storage class.
 
     """
 
     if cls == 'remote':
         from .api.client import RemoteStorage as Storage
     elif cls == 'local':
         from .storage import Storage
+    elif cls == 'memory':
+        from .in_memory import Storage
     else:
         raise ValueError('Unknown storage class `%s`' % cls)
 
     return Storage(**args)