Fix compile issues from last build... whoops