Tweaked the fix by adding new method instead of new parameter