Fix for #90 (closed)
A few lines added analogous to existing code. Please check whether this is appropriate.
Just run pytest.
For further good practices have a look at our review guidelines.