fixed false positive error in test
the test was testing the wrong class in test case