Fix Amplitude tests