Fix parameters in refactored method