fix stupid unit bug