Fix compile error on test

This is why we test shit!