Fixing bug I introduced