Added a class injection assertion