Fix import statement in pulses tests.