Fix stageMask (0x1) usage in various tests