Transport theory with non-local corrections