Fixing minor bug introduced in last commit.