test.py: Fix broken setup function