Deprecation inializer

I’m getting a bunch of deprecation warnings about using the initializer API:

DEPRECATION: lookup was called on a Registry. The initializer API no longer receives a container, and you should use an instanceInitializer to look up objects from the container.

There’s an example in the deprecation guide on how to fix this, but how would I fix the following?

export function initialize(container, application) {
    application.inject('controller', 'heyman', 'heyman:main');
}

export default {
    name: 'setup-heyman',
    initialize
};

Here is a commit I did to support ember 1.13/ and ember 2.0 in a single addon

https://github.com/toranb/ember-cli-simple-store/commit/1f68d63e445a83b9298d03c9bbf95af7b93680d7