Fix bug in GPU list balancing