Removed bug

Squashed a bug. Hope there are no more bugs here.