Update script to behave as expected.