Change signature for test