Possible memory issue fix

Forgot these two in the previous commit.