BUG: consistent subplot ax handling