Update function signature.

Looks like a refactoring leftover.