Add back in discovery gem