[TASK] Adjust to site package