Fix test to allow builds