Ahh nevermind. The DI works with this fine.