Replaced mocks with a simple feature check.