Changing API to reflect latest mne-python.