Bug: React will blur on every key press when this HOC is used. Due to the component returned not being the same?

Usage.js

compose(
    ownState(
        set => ({
            setValue: e => set(state => ({
                ...state,
                value: e.target.value,
            })),
            clearValue: () => set(state => ({
                ...state,
                value: '',
            }))
        }),
        { value: '' },
    ),
    updateStore(
        store,
        set => ({
            saveTodo: description =>
                set(state => ({
                    ...state,
                    todos: [
                        ...state.todos,
                        { description }
                    ],
                })),
        })
    ),
    withProps(({ saveTodo, clearValue }) =>
        keyMap({
            Enter: e => {
                saveTodo(e.target.value);
                clearValue();
            },
            Escape: clearValue,
        })
    ),
    propMap(
        props => ({
            ...props,
            onChange: props.setValue,
        })
    )
)(TextInput);

WithProps.js

import React from 'react';
export default cb => Component => {
    class WithPropsHOC extends React.PureComponent {
        render() {
            const props = this.props;
            const Child = cb(props)(Component);
            return <Child {...props} />
        }
    }

    return WithPropsHOC;
};