Fix wrap of class Energy.