UpdateUnit Bug Fix,