Applied small optimization

To prevent redundancy