Fixed a oversight

I think that is what you wanted to do there ?

*The next pull request will be larger btw ;)*