Improved the patch by following Goober5000's advice.
Missed a line.