Fix invariant with getInitialState