Two small changes to result