Small input fix