1 files changed, 5 insertions, 6 deletions

diff --git a/php-malware-finder/phpmalwarefinder b/php-malware-finder/phpmalwarefinder
index 98c578d..99d9033 100755
--- a/php-malware-finder/phpmalwarefinder
+++ b/php-malware-finder/phpmalwarefinder
@@ -31,13 +31,12 @@ fi
 # before starting yara, check if the file
 one_line_trick() {
 
-    find $@ -type f -print0  | while IFS= read -r -d '' -r file; do
-     line_num=$(wc -l "$file" | cut -d' ' -f1)
-     char_num=$(wc -c "$file" | cut -d' ' -f1)
-
-     if [ "$line_num" -le "2" ]; then
+    find $@ -type f -iname '*.ph*' -print0  | while IFS= read -r -d '' -r file; do
+     read lines words chars name <<< $(wc "$file")
+     
+     if [ "$lines" -le "2" ]; then
         # humm, 2 lines long file ?
-        if [ "$char_num" -ge "300" ]; then
+        if [ "$chars" -ge "300" ]; then
             echo TooShort $file
         fi;
     fi;