Update to support Spring DI