diff --git a/inventory-app/oidc.py b/inventory-app/oidc.py
index bbdbf9f..17106b2 100644
--- a/inventory-app/oidc.py
+++ b/inventory-app/oidc.py
@@ -47,8 +47,10 @@ def add_url_params(url, params):
 
 def add_session_tokens(session, r):
     session["access_token"] = r["access_token"]
-    session["id_token"] = r["id_token"]
-    session["refresh_token"] = r["refresh_token"]
+    if "id_token" in r:
+        session["id_token"] = r["id_token"]
+    if "refresh_token" in r:
+        session["refresh_token"] = r["refresh_token"]
 
 @page_oidc.route('/login-callback')
 def login_callback():