Just a small clean up