Fix issue, minor update