Minor update to correct a memory issue