Fixed build errors

I was getting build errors when I tried to use this addon, I think this should fix them...