Didn't add include

Small fix