Fix script to 2.7