fixup! DI fix
Update build