Fixed the testcase as it was calling wrong method