Clarifying code

A little simplification in the code.