Update test to work with single param toolbox method.