diff --git a/pkg/collector/metric/node_metric.go b/pkg/collector/metric/node_metric.go
index 730caecbfd..cdd7782ca7 100644
--- a/pkg/collector/metric/node_metric.go
+++ b/pkg/collector/metric/node_metric.go
@@ -398,7 +398,7 @@ func (ne *NodeMetrics) UpdateDynEnergy() {
 
 func (ne *NodeMetrics) CalcDynEnergy(component, id string) {
 	total := ne.getAbsoluteEnergyStatCollection(component).Stat[id].Delta
-	fmt.Printf("Energy stat: %v (%s)", ne.getIdleEnergyStatCollection(component).Stat, id)
+	klog.V(5).Infof("Energy stat: %v (%s)", ne.getIdleEnergyStatCollection(component).Stat, id)
 	idle := ne.getIdleEnergyStatCollection(component).Stat[id].Delta
 	dyn := calcDynEnergy(total, idle)
 	ne.getDynEnergyStatCollection(component).SetDeltaStat(id, dyn)
diff --git a/pkg/model/model.go b/pkg/model/model.go
index fd8bed0176..3ad44d0c78 100644
--- a/pkg/model/model.go
+++ b/pkg/model/model.go
@@ -134,6 +134,7 @@ func createPowerModelEstimator(modelConfig *types.ModelConfig) (PowerMoldelInter
 			FloatFeatureNames:           featuresNames,
 			SystemMetaDataFeatureNames:  modelConfig.SystemMetaDataFeatureNames,
 			SystemMetaDataFeatureValues: modelConfig.SystemMetaDataFeatureValues,
+			EnergySource:                modelConfig.EnergySource,
 		}
 		err := model.Start()
 		if err != nil {