added missing class in testing code