Fixing a bug I introduced in the prev commit