FIX: Fix test
Merge pull request #313 from Eric89GXL/ipynb-fix

FIX: Fix test