Bug fix: Fixing mismatched method parameters