Test fix

Two tests were using old Pipeline classes and broke after the latest Pipe updates.