Add input fix