2 small changes