Updated class name of test_sample.py