fix a bug in the assembly
fix test outputs