Fixing control script.