Fixed a bug where units weren't properly removed