- Added missing test class