why dont we need ot change the STI type? not sure if i fully follow the logic (as we are missing the tests)