Small optimization

If it's not too late for next upload...