Fixed 'Energy' Bug

Added condition.