fixed false positive error in test

the test was testing the wrong class in test case