Tiny fix for the latest changes.