Fixed compile errors (oops)