Fixed the DI configuration

This fixes a mistakes introduced in #24.
Closes #25