Fix input script issue