diff --git a/Specification.EntityFrameworkCore/src/Ardalis.Specification.EntityFrameworkCore/RepositoryBaseOfT.cs b/Specification.EntityFrameworkCore/src/Ardalis.Specification.EntityFrameworkCore/RepositoryBaseOfT.cs
index 3ab92ebf..f36da477 100644
--- a/Specification.EntityFrameworkCore/src/Ardalis.Specification.EntityFrameworkCore/RepositoryBaseOfT.cs
+++ b/Specification.EntityFrameworkCore/src/Ardalis.Specification.EntityFrameworkCore/RepositoryBaseOfT.cs
@@ -37,7 +37,7 @@ public virtual async Task<T> AddAsync(T entity, CancellationToken cancellationTo
         /// <inheritdoc/>
         public virtual async Task UpdateAsync(T entity, CancellationToken cancellationToken = default)
         {
-            dbContext.Entry(entity).State = EntityState.Modified;
+            dbContext.Set<T>().Update(entity);
 
             await SaveChangesAsync(cancellationToken);
         }