Minor modification of non-publich script