Small bit of cleanup