Fix slight issue with previous commit.