Fixing bad code for Pipeline test