Fixed a couple build errors