Fixing if newState is terminal