Fixed existing test