Fix to balls demo