diff --git a/llm/data_gemma/datacommons.py b/llm/data_gemma/datacommons.py
index f118060..5b36cf8 100644
--- a/llm/data_gemma/datacommons.py
+++ b/llm/data_gemma/datacommons.py
@@ -11,7 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 """Data Commons."""
 
 import concurrent.futures
@@ -24,7 +23,6 @@
 from data_gemma import base
 from data_gemma import utils
 
-
 _BASE_URL = 'https://{env}.datacommons.org/nodejs/query'
 
 _POINT_MODE = 'toolformer_rig'
@@ -45,7 +43,7 @@ def __init__(
       api_key: str,
       verbose: bool = True,
       num_threads: int = 10,
-      env: str = 'dev',
+      env: str = 'datagemma',
       session: requests.Session | None = None,
   ):
     self.options = base.Options(verbose=verbose)