Bug fix

What was I thinking?!