Fixing test.

Changed the call