Fix state and Balls bug