Minor bug fix for last commit.