Fix first bug in test/test.py.