Fix forgotten unit entry