Fix script one more time