minor function change