diff --git a/vagrant/bootstrap.sh b/vagrant/bootstrap.sh
index 62fac13..0425166 100644
--- a/vagrant/bootstrap.sh
+++ b/vagrant/bootstrap.sh
@@ -2,11 +2,12 @@
 
 set -e
 
-if [ -d /vagrant/colab ]; then
-  basedir=/vagrant/colab
-else
-  basedir=/vagrant
-fi
+for dir in /vagrant/colab /vagrant; do
+    if [ -f $dir/setup.py ]; then
+        basedir="$dir"
+        break
+    fi
+done
 
 # very simple OS detection
 if [ -x /usr/bin/apt-get ]; then
--
libgit2 0.21.2