Fix for only one element in split_button

Signed-off-by: Guillaume Poirier-Morency <guillaumepoiriermorency@gmail.com>