small fix (add param)