Slight change to not allocate unnecessarily

Getting educated by @jpfuentes2