Adding the missed test class