Fix test for new calling convention