Fix compilation problem :(