change module name from AST to Shift in tests