Fixing compilation problem