Updating due to pretty_midi changes