Fixed issue in previous commit

Forgot to save these changes in my previous commit (oops)