Code review changes

Author: ferozk

src/commands/element.ts

@@ -33,15 +33,15 @@ export async function findElOrEls(
             ? JSON.parse(selector)
             : selector; // Special case
 
+      // If user is looking for Native IOS/Mac locator
       if (
-         proxyDriver instanceof AndroidUiautomator2Driver ||
-         proxyDriver instanceof XCUITestDriver
+         !isFlutterLocator &&
+         (proxyDriver instanceof XCUITestDriver ||
+            proxyDriver instanceof Mac2Driver)
       ) {
-         return { strategy, selector, context };
-      } else if (isFlutterLocator) {
-         return { using: strategy, selector: selector, context };
-      } else {
          return { using: strategy, value: parsedSelector, context };
+      } else {
+         return { strategy, selector: parsedSelector, context };
       }
    }
 

test/unit/element.specs.ts

@@ -156,7 +156,7 @@ describe('Element Interaction Functions', () => {
          expect(
             mockDriver.command.calledWith('/element', 'POST', {
                strategy: 'strategy',
-               selector: 'selector',
+               value: 'selector',
                context: 'context',
             }),
          ).to.be.true;