Comments and og_set_theme

Fixed problem in og_user_roles_theme() where og_set_theme() was returning the wrong theme on comment/edit.

Needed this code:

if (arg(1) == 'edit') {
$comment = _comment_load(arg(2));
$nid = $comment->nid;
$gid = og_user_roles_getgid($nid, $uid);
$group_node = og_set_theme($gid);
} else {
$group_node = og_set_theme(arg(2));
}