Fix compile... Unit.h