[FIX] add context safe guard