diff --git a/litellm/tests/test_utils.py b/litellm/tests/test_utils.py
index 2801cd6d6d..aab86ac4ed 100644
--- a/litellm/tests/test_utils.py
+++ b/litellm/tests/test_utils.py
@@ -83,20 +83,13 @@ def test_get_valid_models():
 
 # test_get_valid_models()
 
-# def test_bad_key():
-#     key = "bad-key"
-#     response = check_valid_key(key)
-#     if response == False:
-#         pass
-#     else:
-#         raise Exception("Bad key was not detected")
-# test_bad_key()
+def test_bad_key():
+    key = "bad-key"
+    response = check_valid_key(model="gpt-3.5-turbo", api_key=key)
+    print(response, key)
+    assert(response == False)
 
-# def test_good_key():
-#     key = os.environ['OPENAI_API_KEY']
-#     response = check_valid_key(key)
-#     if response == True:
-#         pass
-#     else:
-#         raise Exception("Good key did not pass")
-# test_good_key()
\ No newline at end of file
+def test_good_key():
+    key = os.environ['OPENAI_API_KEY']
+    response = check_valid_key(model="gpt-3.5-turbo", api_key=key)
+    assert(response == True)
diff --git a/pyproject.toml b/pyproject.toml
index 716b62615d..306a0a866d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm"
-version = "0.1.604"
+version = "0.1.605"
 description = "Library to easily interface with LLM API providers"
 authors = ["BerriAI"]
 license = "MIT License"