1 files changed, 8 insertions, 1 deletions
diff --git a/bin/symlinks b/bin/symlinks
index c331261..a2f27cf 100755
--- a/bin/symlinks
+++ b/bin/symlinks
@@ -15,6 +15,13 @@ elif [ ! -e "$MODULES" ]; then
   echo "Not found: $MODULES"
 fi
 
+# Set puppet folder
+if [ -d "puppet" ]; then
+  PUPPET_FOLDER="puppet"
+else
+  PUPPET_FOLDER="."
+fi
+
 # Add module symlinks using absolute folders
 for module in `ls $MODULES`; do
   if [ "$module" == "bootstrap" ]; then
@@ -26,5 +33,5 @@ for module in `ls $MODULES`; do
   fi
 
   path="`cd $MODULES/$module && pwd`"
-  ( cd modules &> /dev/null && ln -sf $path )
+  ( cd $PUPPET_FOLDER/modules &> /dev/null && ln -sf $path )
 done