Fix comppilation issue