Fixed a bug in my previous commit.