Fix #14

I did what the error told me to.
Not sure if this is helpful.
Take it or leave it.