Fix Build

Not sure how it builds for you guys, but I couldn't get it to build without these changes...