Fix ps5000a import