TODO:  add this to util.py or drop?