Fix partition script