Added compatibility with yt 3.0