Fix to be able to compile