fix: use diff context instead of prop drilling