Now the tests works for this class.