Minor optimization of code.