Minor coding fix