Actually fixed bug from previous commit.