Fixed bug from prev commit.