Fixed a bug with the previous commit.