Parameter type fix

Fixed warnings due to slightly mismatching parameter types.