Fix small bug in previous commit ...

... sorry, not enough tested previously :(