Fix version change in test. Get tests passing