Updated to match new API for updating current assignment location.