Fix indentation in test.py