This is only used in tests, right? But I assume it must be defined within the main module and not in the test?