Fix reference to main script