fix code based on fail test run

Benjamin Mako Hill || Want to submit a patch?