Make argument to step_func_done optional

Useful if you're expecting an event or callback but don't want to assert stuff.