fix tests for inject_inputs