Fix script issue