Small script change