diff --git a/src/utils/set.ts b/src/utils/set.ts
index e48ffe55..bef27073 100644
--- a/src/utils/set.ts
+++ b/src/utils/set.ts
@@ -24,7 +24,13 @@ function internalSet<Entity = any, Output = Entity, Value = any>(
   }
 
   // Delete prop if `removeIfUndefined` and value is undefined
-  if (removeIfUndefined && value === undefined && restPath.length === 1) {
+  if (removeIfUndefined && value === undefined && restPath.length === 0) {
+    delete clone[path];
+  } else if (
+    removeIfUndefined &&
+    value === undefined &&
+    restPath.length === 1
+  ) {
     const origin = clone[path];
     if (origin && typeof origin === 'object') {
       const child = Array.isArray(origin) ? [...origin] : { ...origin };
diff --git a/tests/utils.test.ts b/tests/utils.test.ts
index 248cb8f0..1d7fc766 100644
--- a/tests/utils.test.ts
+++ b/tests/utils.test.ts
@@ -97,6 +97,31 @@ describe('utils', () => {
       expect('lv4' in longTgt.lv1.lv2.lv3).toBeFalsy();
     });
 
+    it.each(['light', Symbol('light')])(
+      'removes a top-level property without changing the source: %s',
+      key => {
+        const source: Record<string | symbol, number> = {
+          [key]: 2333,
+          bamboo: 1,
+        };
+        const target = set(source, [key], undefined, true);
+
+        expect(target).toEqual({ bamboo: 1 });
+        expect(Object.prototype.hasOwnProperty.call(target, key)).toBe(false);
+        expect(source[key]).toBe(2333);
+      },
+    );
+
+    it('removes a top-level array entry without shifting other entries', () => {
+      const source = ['light', 'bamboo'];
+      const target = set(source, [0], undefined, true);
+
+      expect(0 in target).toBe(false);
+      expect(target[1]).toBe('bamboo');
+      expect(target).toHaveLength(2);
+      expect(source).toEqual(['light', 'bamboo']);
+    });
+
     it.each(['str', 123, true, Symbol('value'), BigInt(1)])(
       'preserves a primitive parent when removing a missing property: %s',
       value => {