fix tests due to changing the public exposure