FIX: make return figure if axes is None

cc @agramfort @mainakjas