Fix compile error

- Don't leave the code in a broken compile state