Fixed the bug from last night.

Removed some leftover code from testing that messed things up.