Better use DI container here?