Fix for derivatives of initial state in the RK comp