Fixing compilation bug