Edited the wrong lines in the last commit, here's the real fix.