Small correction for compilation