Fix error with function signature