Quickfix on test.py