Change behavior and signature of registerWithDIContainer()