Fixed compile errors (oops!)