Fixed tests after signature change in tests