Code optimization

I made a code optimization to make it cleaner. =)