React Apollo Error: Invariant Violation: Impossible de trouver "client" dans le contexte ou transmis en option

Je construis un projet en utilisant React, Apollo et Next.js. J'essaye de mettre à jour react-apollo vers 3.1.3 et j'obtiens maintenant l'erreur suivante lors de la visualisation du site.

Violation invariante: Impossible de trouver "client" dans le contexte ou transmis en option. Enveloppez le composant racine dans une ou passez une instance ApolloClient dans via des options.

Si je rétrograde le paquet react-apollo vers 2.5.8, cela fonctionne sans problème donc je pense que quelque chose a changé entre 2.5 et 3.x mais je ne trouve rien dans la documentation react-apollo ou next-with-apollo pour indiquer ce que cela pourrait être. Toute assistance sera grandement appréciée.

withData.js

import App from 'next/app';
import { ApolloProvider } from 'react-apollo';
import Page from '../components/Page';
import { Overlay } from '../components/styles/Overlay';
import withData from '../lib/withData';

class MyApp extends App {
    static async getInitialProps({ Component, ctx }) {
        let pageProps = {};
        if (Component.getInitialProps) {
            pageProps = await Component.getInitialProps(ctx);
        }

        // this exposes the query to the user
        pageProps.query = ctx.query;
        return { pageProps };
    }

    render() {
        const { Component, apollo, pageProps } = this.props;

        return (
            <ApolloProvider client={apollo}>
                <Overlay id="page-overlay" />
                <Page>
                    <Component {...pageProps} />
                </Page>
            </ApolloProvider>
        );
    }
}

export default withData(MyApp);

_app.js

import withApollo from 'next-with-apollo';
import ApolloClient from 'apollo-boost';
import { endpoint } from '../config';

function createClient({ headers }) {
    return new ApolloClient({
        uri: endpoint,
        request: operation => {
            operation.setContext({
                fetchOptions: {
                    credentials: 'include'
                },
                headers
            });
        },
        // local data
        clientState: {
            resolvers: {
                Mutation: {}
            },
            defaults: {}
        }
    });
}

export default withApollo(createClient);