Fix for checkout branch issue