Minor coding organization

Just noticed XD