minor fix for entity generation