1 files changed, 4 insertions, 13 deletions
diff --git a/src/net/network.cpp b/src/net/network.cpp
index 9e6f6621..366d07d9 100644
--- a/src/net/network.cpp
+++ b/src/net/network.cpp
@@ -182,17 +182,9 @@ void Network::registerHandler(MessageHandler *handler)
 
 void Network::unregisterHandler(MessageHandler *handler)
 {
-    const Uint16 *i = handler->handledMessages;
-
-    while(*i)
+    for (const Uint16 *i = handler->handledMessages; *i; i++)
     {
-        std::map<Uint16, MessageHandler*>::iterator iter;
-        iter = mMessageHandlers.find(*i);
-        if (iter != mMessageHandlers.end())
-        {
-            mMessageHandlers.erase(iter);
-        }
-        i++;
+        mMessageHandlers.erase(*i);
     }
 
     handler->setNetwork(0);
@@ -200,7 +192,7 @@ void Network::unregisterHandler(MessageHandler *handler)
 
 void Network::clearHandlers()
 {
-    std::map<Uint16, MessageHandler*>::iterator i;
+    MessageHandlerIterator i;
     for (i = mMessageHandlers.begin(); i != mMessageHandlers.end(); i++)
     {
         i->second->setNetwork(0);
@@ -214,8 +206,7 @@ void Network::dispatchMessages()
     {
         MessageIn msg = getNextMessage();
 
-        std::map<Uint16, MessageHandler*>::iterator iter;
-        iter = mMessageHandlers.find(msg.getId());
+        MessageHandlerIterator iter = mMessageHandlers.find(msg.getId());
 
         if (iter != mMessageHandlers.end())
             iter->second->handleMessage(&msg);