Added a small optimization that should make the tool run smoother.

Thanks to brunoais for finding this optimization.