Little fix

This hack is not really clean, but seems to work.