Fix balance in initial data