Updating the method call after the merge of 1332 due of refactor