diff --git a/tests/shells/test_bash.py b/tests/shells/test_bash.py
index 117e3af4..b10ae0e3 100644
--- a/tests/shells/test_bash.py
+++ b/tests/shells/test_bash.py
@@ -51,6 +51,7 @@ class TestBash(object):
     def test_app_alias_variables_correctly_set(self, shell):
         alias = shell.app_alias('fuck')
         assert "fuck () {" in alias
+        assert 'TF_SHELL=bash' in alias
         assert "TF_ALIAS=fuck" in alias
         assert 'PYTHONIOENCODING=utf-8' in alias
         assert 'TF_SHELL_ALIASES=$(alias)' in alias
diff --git a/tests/shells/test_zsh.py b/tests/shells/test_zsh.py
index e31c362e..fc20a061 100644
--- a/tests/shells/test_zsh.py
+++ b/tests/shells/test_zsh.py
@@ -51,6 +51,7 @@ class TestZsh(object):
     def test_app_alias_variables_correctly_set(self, shell):
         alias = shell.app_alias('fuck')
         assert "fuck () {" in alias
+        assert 'TF_SHELL=zsh' in alias
         assert "TF_ALIAS=fuck" in alias
         assert 'PYTHONIOENCODING=utf-8' in alias
         assert 'TF_SHELL_ALIASES=$(alias)' in alias