--- a/src/energy/test/basic-energy-model-test.cc	Fri Sep 05 15:38:55 2014 -0700
+++ b/src/energy/test/basic-energy-model-test.cc	Fri Sep 05 16:33:57 2014 -0700
@@ -93,6 +93,7 @@
   NS_TEST_ASSERT_MSG_EQ (StateSwitchTest (WifiPhy::TX), false, "Problem with state switch test (WifiPhy tx).");
   NS_TEST_ASSERT_MSG_EQ (StateSwitchTest (WifiPhy::RX), false, "Problem with state switch test (WifiPhy rx).");
   NS_TEST_ASSERT_MSG_EQ (StateSwitchTest (WifiPhy::SWITCHING), false, "Problem with state switch test (WifiPhy switching).");
+  NS_TEST_ASSERT_MSG_EQ (StateSwitchTest (WifiPhy::SLEEP), false, "Problem with state switch test (WifiPhy sleep).");
 }
 
 bool
@@ -178,6 +179,9 @@
     case WifiPhy::SWITCHING:
       current = devModel->GetSwitchingCurrentA ();
       break;
+    case WifiPhy::SLEEP:
+      current = devModel->GetSleepCurrentA ();
+      break;
     default:
       NS_FATAL_ERROR ("Undefined radio state: " << state);
       break;