Fix minor bug (not supplying all arguments to new function)