2 files changed, 5 insertions, 4 deletions
diff --git a/MdeModulePkg/Universal/Network/SnpDxe/Snp.c b/MdeModulePkg/Universal/Network/SnpDxe/Snp.c
index e90e4e0244..2dc0536117 100644
--- a/MdeModulePkg/Universal/Network/SnpDxe/Snp.c
+++ b/MdeModulePkg/Universal/Network/SnpDxe/Snp.c
@@ -1,7 +1,7 @@
 /** @file
   Implementation of driver entry point and driver binding protocol.
 
-Copyright (c) 2004 - 2010, Intel Corporation. All rights reserved.<BR>
+Copyright (c) 2004 - 2011, Intel Corporation. All rights reserved.<BR>
 This program and the accompanying materials are licensed
 and made available under the terms and conditions of the BSD License which
 accompanies this distribution. The full text of the license may be found at
@@ -1014,7 +1014,7 @@ InitializeSnpNiiDriver (
            ImageHandle,
            SystemTable,
            &mSimpleNetworkDriverBinding,
-           NULL,
+           ImageHandle,
            &gSimpleNetworkComponentName,
            &gSimpleNetworkComponentName2
            );
diff --git a/MdeModulePkg/Universal/Network/SnpDxe/SnpDxe.inf b/MdeModulePkg/Universal/Network/SnpDxe/SnpDxe.inf
index 70dd3eeca7..a7f81c83c5 100644
--- a/MdeModulePkg/Universal/Network/SnpDxe/SnpDxe.inf
+++ b/MdeModulePkg/Universal/Network/SnpDxe/SnpDxe.inf
@@ -1,7 +1,7 @@
 ## @file
 #    Component description file for SNP module.
 #
-# Copyright (c) 2006 - 2010, Intel Corporation. All rights reserved.<BR>
+# Copyright (c) 2006 - 2011, Intel Corporation. All rights reserved.<BR>
 # This program and the accompanying materials are licensed 
 # and made available under the terms and conditions of the BSD License which 
 # accompanies this distribution. The full text of the license may be found at 
@@ -19,6 +19,7 @@
   MODULE_TYPE                    = UEFI_DRIVER
   VERSION_STRING                 = 1.0
   ENTRY_POINT                    = InitializeSnpNiiDriver
+  UNLOAD_IMAGE                   = NetLibDefaultUnload
 
 #
 # The following information is for reference only and not required by the build tools.
@@ -63,7 +64,7 @@
   UefiDriverEntryPoint
   BaseMemoryLib
   DebugLib
-
+  NetLib
 
 [Guids]
   gEfiEventExitBootServicesGuid                 ## CONSUMES