Update script to reflect changes in KeyLocator