diff --git a/.gitignore b/.gitignore
index fada528..ceb6ffe 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,3 +2,5 @@
 node_modules/
 dist/
 .idea
+.vscode
+.yarn
\ No newline at end of file
diff --git a/src/omitByKeys.ts b/src/omitByKeys.ts
index 4b7e6c3..9e5161f 100644
--- a/src/omitByKeys.ts
+++ b/src/omitByKeys.ts
@@ -6,6 +6,7 @@
  */
 
 const omitByKeys = (originalObject: object, keys: string | string[]) => {
+  keys = Array.isArray(keys) ? keys : [keys]
   const newObj = {}
   const objKeys = Object.keys(originalObject)
   for (let i = 0; i < objKeys.length; i++) {
diff --git a/src/pickByKeys.ts b/src/pickByKeys.ts
index 3c8b75d..95f5bc8 100644
--- a/src/pickByKeys.ts
+++ b/src/pickByKeys.ts
@@ -6,6 +6,7 @@
  */
 
 const pickByKeys = (originalObject: object, keys: string | string[]) => {
+  keys = Array.isArray(keys) ? keys : [keys]
   const newObj = {}
   for (let i = 0; i < keys.length; i++) {
     newObj[keys[i]] = originalObject[keys[i]]