1 files changed, 2 insertions, 4 deletions
diff --git a/xmpp-vala/src/module/roster/flag.vala b/xmpp-vala/src/module/roster/flag.vala
index 43d185d9..2e93911e 100644
--- a/xmpp-vala/src/module/roster/flag.vala
+++ b/xmpp-vala/src/module/roster/flag.vala
@@ -1,14 +1,12 @@
 using Gee;
 
-using Xmpp.Core;
-
 namespace Xmpp.Roster {
 
 public class Flag : XmppStreamFlag {
     public const string ID = "roster";
     public static FlagIdentity<Flag> IDENTITY = new FlagIdentity<Flag>(NS_URI, ID);
 
-    public HashMap<string, Item> roster_items = new HashMap<string, Item>();
+    public HashMap<Jid, Item> roster_items = new HashMap<Jid, Item>();
 
     public string? iq_id;
 
@@ -16,7 +14,7 @@ public class Flag : XmppStreamFlag {
         return roster_items.values;
     }
 
-    public Item? get_item(string jid) {
+    public Item? get_item(Jid jid) {
         return roster_items[jid];
     }