fixed code error due to mahtod signature change in tests