Update with input fix