fix small unit null bug thing