Minor logic fix to last commit.