Trying workarounds in test.py