Fix for incorrect parameter type