Fix to make this script consistent with new slot API