Add gemma's selection back in