Remove NotImplementedError