Make sure you squish the rice balls, firmly. If it does not prevent the rice balls falling apart, You can add Katakuriko (potato starch) 1 tbsp after the rice is cooked. Stir it well into the rice, that will keep the rice sticking together.