RunStates were not being propagated correctly on functions (function.apply)