diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index 31ca9a35..fa9478bd 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -4,8 +4,6 @@ on:
     branches:
       - main
   push:
-    branches:
-      - main
 jobs:
   test:
     runs-on: ubuntu-latest
@@ -29,6 +27,9 @@ jobs:
           environment-file: environment.yml
           python-version: 3.7
           activate-environment: rms_env
+          miniforge-variant: Mambaforge
+          miniforge-version: latest
+          use-mamba: true
       - name: Fix PyCall linking
         run: |
             julia -e 'ENV["PYTHON"]=ENV["CONDA"] * "/envs/rms_env/bin/python"; using Pkg; Pkg.add("PyCall"); Pkg.build("PyCall")'
@@ -42,4 +43,4 @@ jobs:
       - uses: julia-actions/julia-processcoverage@v1
       - uses: codecov/codecov-action@v1
         with:
-          file: lcov.info
+          file: lcov.info
\ No newline at end of file
diff --git a/Project.toml b/Project.toml
index 6236e221..8fd3639d 100644
--- a/Project.toml
+++ b/Project.toml
@@ -43,14 +43,14 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 Calculus = "0.4,0.5"
 Colors = "0.11,0.12"
 Conda = "1"
-DiffEqSensitivity = "6"
+DiffEqSensitivity = "^6"
 ForwardDiff = "0.10"
 Images = "0.24"
 IncompleteLU = "0.2"
 IterTools = "1.3"
 LsqFit = "0.12"
 ModelingToolkit = "8"
-OrdinaryDiffEq = "6"
+OrdinaryDiffEq = "^6"
 Parameters = "0.12"
 PreallocationTools = "0"
 PyCall = "1"
@@ -58,7 +58,7 @@ PyPlot = "2"
 QuartzImageIO = "0.7"
 RecursiveArrayTools = "2.17"
 ReverseDiff = "1.9"
-SciMLBase = "1"
+SciMLBase = "^1"
 SmoothingSplines = "0.3"
 SpecialFunctions = "1"
 StaticArrays = "1"
diff --git a/environment.yml b/environment.yml
index af867cc9..55107726 100644
--- a/environment.yml
+++ b/environment.yml
@@ -2,8 +2,8 @@ name: rms_env
 channels:
   - defaults
   - rmg
+  - jacksonburns
 dependencies:
-  - rdkit
-  - rmg
+  - rmgmolecule
   - pydot
-  - yaml
+  - matplotlib