Modify scaffolding process in test