Fixes discovery unit test