Little fix

This hack is not really clean, but seems to work.
Merge pull request #1 from florentmorin/patch-1

Little fix