Made test_contracts.py work