NotImplementedError bug fix.