Fixing issue with get_current_branch