Fix push code, so it will work on any branches.