diff --git a/environment-windows.yml b/environment-windows.yml
index 55f0761..83aee4a 100644
--- a/environment-windows.yml
+++ b/environment-windows.yml
@@ -27,22 +27,23 @@ dependencies:
   - scikit-learn=0.22
   - scipy=1.3.1
   - tqdm=4.40.0
-  - tensorflow=2.0.0 # or tensorflow-gpu if gpu
   - tensorboard=2.0.0
+  - tensorflow=2.0.0 # or tensorflow-gpu if gpu
   - tensorflow-datasets=1.2.0
   - tensorflow-estimator=2.0.0
   - tensorflow-hub=0.7.0
-  - tensorflow-metadata=0.14.0
   - tensorflow-probability=0.7
   - wheel
   - widgetsnbextension=3.5.1
   - pip:
-    #- atari_py==0.2.6 # NOT ON WINDOWS YET
+    #- atari-py==0.2.6 # NOT ON WINDOWS YET
     - gym==0.15.4
     - opencv-python==4.1.2.30
     - psutil==5.6.7
     - pyglet==1.3.2
+    #- tensorflow-addons==0.6.0 # NOT ON WINDOWS YET
     #- tensorflow-data-validation==0.15.0 # NOT ON WINDOWS YET
+    #- tensorflow-metadata==0.15.1 # NOT ON WINDOWS YET
     #- tensorflow-model-analysis==0.15.4 # NOT ON WINDOWS YET
     - tensorflow-serving-api==2.0.0 # or tensorflow-serving-api-gpu if gpu
     #- tensorflow-transform==0.15.0 # NOT ON WINDOWS YET
diff --git a/environment.yml b/environment.yml
index b8023e0..12f3ea4 100644
--- a/environment.yml
+++ b/environment.yml
@@ -29,12 +29,9 @@ dependencies:
   - tqdm=4.40.0
   - tensorboard=2.0.0
   - tensorflow=2.0.0 # or tensorflow-gpu if gpu
-  - tensorflow-addons=0.6.0
-  - tensorflow-base=2.0.0
   - tensorflow-datasets=1.2.0
   - tensorflow-estimator=2.0.0
   - tensorflow-hub=0.7.0
-  - tensorflow-metadata=0.15.1
   - tensorflow-probability=0.7
   - wheel
   - widgetsnbextension=3.5.1
@@ -44,7 +41,9 @@ dependencies:
     - opencv-python==4.1.2.30
     - psutil==5.6.7
     - pyglet==1.3.2
+    - tensorflow-addons==0.6.0
     - tensorflow-data-validation==0.15.0
+    - tensorflow-metadata==0.15.1
     - tensorflow-model-analysis==0.15.4
     - tensorflow-serving-api==2.0.0 # or tensorflow-serving-api-gpu if gpu
     - tensorflow-transform==0.15.0