diff --git a/generate.sh b/generate.sh
index e6f81456..c49406b5 100755
--- a/generate.sh
+++ b/generate.sh
@@ -44,7 +44,7 @@ done
 
 # count the number of IP
 cd result
-ls | 
+ls *.txt | 
 while read file; do 
 	echo ${file%.*}
 	cat $file |