Fix a bug caused by the last commit.

It was actually a bug that I broke by fixing the bug in the last commit, it's bugception!