follow up on the bug from the previous commit.