Remove state from HOC