Fix missing last unit