chore: clean up entity tests