use _unset_env_vars during teardown

fwojciec · Oct 24, 2020 · 3dd017e · 3dd017e
1 parent 061f45b
commit 3dd017e
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/dsemu/wrapper.py b/dsemu/wrapper.py
@@ -66,11 +66,9 @@ def stop(self) -> None:
         """
         if self._instance is not None and self._is_healthy():
             self._request(SHUTDOWN_ENDPOINT, method="POST")
-            os.unsetenv("DATASTORE_EMULATOR_HOST")
-            os.unsetenv("DATASTORE_PROJECT_ID")
+            self._unset_env_vars()
             self._instance.terminate()
             self._instance = None
-
         self._host = None
         self._project_id = None
         self._gcloud = None