correct issue in entity generation