fix bug in previous commit.