diff --git a/gen-snapshot.sh b/gen-snapshot.sh
index 0321a0b67..9c7e511c9 100755
--- a/gen-snapshot.sh
+++ b/gen-snapshot.sh
@@ -6,7 +6,7 @@ VERSION=2.4.0~$(date "+%Y%m%d")
 BUILDDEST=/tmp/airtime-${VERSION}/
 DEBDIR=`pwd`/debian
 
-git checkout 2.4.x
+git checkout $(git branch | grep "^*" | cut -d' ' -f 2)
 git pull
 
 echo "cleaning up previous build..."