Fix unit test

Used wrong parameter; unit test will pass now.