Fix another bug with components in potential energy computation