Update script to handle 6.0.1