Fixed an issue with the previous commit.