Fixing bug left in previous commit.