Fixed compile errors. Whoops!