Fixed a quick error

Missed a spot for code change, and fixed it now.