More bug fixage