Small cleanup

Right, fixed.