fix: broken test to reflect interface changes.