modify the discovery implementation on controller