- Updated test class