Replaced _class with .class in DI