Entity Module setup fix