Fix incorrect number of arguments in discovery_test.py