Handle { () => () } correctly