Comment out infinite loop in test.py