var objectContext = ((IObjectContextAdapter)context).ObjectContext;
                            ObjectStateEntry entry;

                            if (objectContext.ObjectStateManager.TryGetObjectStateEntry(Key, out entry))
                            {
                                // Re-attach if necessary
                                if (entry.State == EntityState.Detached)
                                {
                                    objectContext.AttachTo(Key.EntitySetName, entry.Entity);
                                }
                                entry.ApplyCurrentValues(Entity);
                            }
                            else
                            {
                                // Attach for the first time
                                context.Set<TEntity>().Attach(Entity);
                                context.Entry(Entity).State = EntityState.Modified;
                            }