X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/1677a24a253d883f3db0744651ea5cb489cbf73a..347f9a2ca55bffc88bf965952b9be3a15535e908:/src/surf/plugins/energy.cpp

diff --git a/src/surf/plugins/energy.cpp b/src/surf/plugins/energy.cpp
index f320560dd7..1ee70cfced 100644
--- a/src/surf/plugins/energy.cpp
+++ b/src/surf/plugins/energy.cpp
@@ -100,7 +100,8 @@ void sg_energy_plugin_init() {
     });
 
     simgrid::surf::VMCreatedCallbacks.connect([](simgrid::surf::VirtualMachine* vm) {
-    	std::map<simgrid::surf::Host*, HostEnergy*>::iterator host_energy_it = surf_energy->find(vm->p_subWs);
+    	std::map<simgrid::surf::Host*, HostEnergy*>::iterator host_energy_it =
+    			surf_energy->find(vm->p_hostPM->extension(simgrid::surf::Host::EXTENSION_ID));
     	xbt_assert(host_energy_it != surf_energy->end(), "The host is not in surf_energy.");
     	(*surf_energy)[vm] = host_energy_it->second;
     	host_energy_it->second->ref(); // protect the HostEnergy from getting deleted too early