Make build work in test