diff --git a/public/help/dandisets.md b/public/help/dandisets.md
index fbb4edd..3d15600 100644
--- a/public/help/dandisets.md
+++ b/public/help/dandisets.md
@@ -4,7 +4,7 @@ Search for and select a dandiset.
 
 <!--------------------------------------------------------------------------------->
 {% if staging %}
-You are viewing the staging site.
+You are viewing the staging site. If you need to access embargoed datasets, configure your API key first.
 
 See [Dandi Archive (staging)](https://gui-staging.dandiarchive.org/).
 {% else %}
diff --git a/python/dendro/compute_resource/register_compute_resource.py b/python/dendro/compute_resource/register_compute_resource.py
index 85a880a..6475786 100644
--- a/python/dendro/compute_resource/register_compute_resource.py
+++ b/python/dendro/compute_resource/register_compute_resource.py
@@ -74,9 +74,9 @@ def register_compute_resource(*, dir: str, compute_resource_id: Optional[str] =
     print(url)
     print('')
 
-    assert compute_resource_id is not None
-    assert compute_resource_private_key is not None
-    return compute_resource_id, compute_resource_private_key
+    assert COMPUTE_RESOURCE_ID is not None
+    assert COMPUTE_RESOURCE_PRIVATE_KEY is not None
+    return COMPUTE_RESOURCE_ID, COMPUTE_RESOURCE_PRIVATE_KEY
 
 def _random_string(length: int) -> str:
     import random