NotImplementedError bugfix