Adding missing test class