Issue #5: Fix role update