Fix bug with first-socket highlight on drop