Fixing bug.

#notetoself - run on local, less coffee when fixing bugs.