Added missing test class