MNT: Fixed demo based on package name change