Fix the tiny script issue