Minor Bug fix

Induced bug from last commit.