Fix first bug in test/test.py.
Fix second bug in test/test.py