Edinburgh Rugby star insists discipline will be improved for Benetton visit