diff --git a/lang_agent/fs_bkends/__init__.py b/lang_agent/fs_bkends/__init__.py
index 988c2fd..302e0bc 100644
--- a/lang_agent/fs_bkends/__init__.py
+++ b/lang_agent/fs_bkends/__init__.py
@@ -2,6 +2,7 @@ import tyro
 
 from lang_agent.fs_bkends.base import BaseFilesystemBackend
 from lang_agent.fs_bkends.statebk import StateBk, StateBkConfig
+from lang_agent.fs_bkends.localshell import LocalShell, LocalShellConfig
 
 statebk_dict = {
     "statebk": StateBkConfig(),