Update fix for endless_data_attack_test.py

This is the fix of issue 80: endless_attack_test fix.