FIX: handle correctly main axis def
fix example according to new API