diff --git a/rocAL/rocAL_pybind/README.md b/rocAL/rocAL_pybind/README.md
index 25308e93e1..7c7c3751f1 100644
--- a/rocAL/rocAL_pybind/README.md
+++ b/rocAL/rocAL_pybind/README.md
@@ -13,9 +13,9 @@ written primarily in C/C++ language can be used effectively in Python.
 * PIP3 - `sudo apt install python3-pip`
 
 ## Install 
-Install rali_pybind using the run.sh script (for all except conda environment)
+Install rocAL_pybind using the run.sh script
 ```
-sudo ./run.sh
+./run.sh
 ```
 NOTE: If using conda environment, use:
 ```
diff --git a/rocAL/rocAL_pybind/run.sh b/rocAL/rocAL_pybind/run.sh
index 7fb24ff8da..fe6d4614c3 100755
--- a/rocAL/rocAL_pybind/run.sh
+++ b/rocAL/rocAL_pybind/run.sh
@@ -1,6 +1,17 @@
-rm -r /usr/local/lib/python3.6/dist-packages/amd_rali-1.1.0-py3.6-linux-x86_64.egg
-rm -r ./amd_rali.egg-info/
-rm -r ./build
-rm -r ./dist
-python3.6 setup.py build
-python3.6 setup.py install
+#!/bin/bash
+
+DEFAULT_PYTHON=$(which python3.6) ## Gets the default python
+CONDA="conda"
+if [ -n "$CONDA_DEFAULT_ENV" ]  || [ -n "$VIRTUAL_ENV" ] || [[ "$DEFAULT_PYTHON" == *"$CONDA"* ]]; then ## Checks if it is in any env then removes packages accordingly
+    PYTHON_LIB_PATH=${DEFAULT_PYTHON/bin/lib}
+    EGG_FILE_PATH="/site-packages/amd_rali-1.1.0-py3.6-linux-x86_64.egg"
+    ROCAL_PYTHON_LIB_PATH=$PYTHON_LIB_PATH$EGG_FILE_PATH
+    sudo rm -r "$ROCAL_PYTHON_LIB_PATH"
+else
+  sudo rm -r /usr/local/lib/python3.6/dist-packages/amd_rali-1.1.0-py3.6-linux-x86_64.egg
+fi
+sudo rm -r ./amd_rali.egg-info/
+sudo rm -r ./build
+sudo rm -r ./dist
+sudo "$DEFAULT_PYTHON" setup.py build
+sudo "$DEFAULT_PYTHON" setup.py install
\ No newline at end of file