Oops...fixed error check