Small bit of clean up