Fix script for this branch too