fix stage2 test
fix coverage