Small bug fix which prevented using an arbitrary filename

[dokeeffe]

Small fix I found while trying with another dataset. Thanks for this repo and videos by the way!!

[jakevdp]

Thanks!