Keep the functions if they are passed in

Looks like one was missed :)