You should also changed all subsequent uses of `expanded_inputs` to use `inputs` instead.