Fix signature in test