Switching to the right unit