Fix unit selection not filling in dimension