Fixed a problem with my previous commit.