fixing missing class/di issue