Fix second bug in test/test.py