fix tests with changed function arguments