[PATCH 4/4] Change internal API for consistency