I'm really tired and can't remember what changes I made. Probably something to do with fixing the multiple unit selections.