[TO_STAGE] Misc bugs