Update unit.cs

seg fault fix