[PATCH 2/2] Refactor SI-related code