added a missing class in Test project