diff --git a/defaults/main.yml b/defaults/main.yml
index ad6560c..1c0da2a 100644
--- a/defaults/main.yml
+++ b/defaults/main.yml
@@ -45,7 +45,7 @@ kpi_python_version: python{{ kpi_python_source_version }}
 kpi_checkout_path: "{{ kpi_versioned_path }}/{{ kpi_git_branch }}"
 kpi_app_path: "{{ kpi_system_user_home }}/app"
 kpi_versioned_path: "{{ kpi_app_path }}-versioned"
-kpi_git_branch: kpi_2.0
+kpi_git_branch: "ona-custom-changes"
 kpi_service_name: "{{ kpi_system_user }}"
 kpi_venv_path: "{{ kpi_system_user_home }}/.virtualenvs/{{ kpi_service_name }}"
 kpi_nvm_install_path: "{{ kpi_system_user_home }}/.nvm"
diff --git a/molecule/default/converge.yml b/molecule/default/converge.yml
index 2ac2d5e..2c85752 100644
--- a/molecule/default/converge.yml
+++ b/molecule/default/converge.yml
@@ -8,4 +8,3 @@
       kpi_pgsql_password: kpi
       kpi_enable_monitoring: false
       kpi_git_url: "https://github.com/onaio/kpi.git"
-      kpi_git_branch: "kpi_2.0"