Fixed bug left in last commit.