Fix unit test to match current method signature (soon to be changed)