diff --git a/dvc/remote/base.py b/dvc/remote/base.py index 90fc167e16..811fd9fe52 100644 --- a/dvc/remote/base.py +++ b/dvc/remote/base.py @@ -75,11 +75,9 @@ def __init__(self, checksum): def index_locked(f): @wraps(f) def wrapper(remote_obj, *args, **kwargs): - remote = kwargs.get("remote") - if remote: - with remote.index: - return f(remote_obj, *args, **kwargs) - return f(remote_obj, *args, **kwargs) + remote = kwargs.get("remote", remote_obj) + with remote.index: + return f(remote_obj, *args, **kwargs) return wrapper