Balance bugfix as it wa. Need to update it to new features.