Fix update check by adding branch if necessary