Fixed regression

v2.0.4 update introduced a regression with arguments. Fixed.