Fix build error

Forgot I moved this before commit