add check for null entity