Bugfix

Whoops! fixed a bug.