Fix logic error with previous commit.