Just a mini code cleanup