Yeah, it’s important to note that heritability is a statistic about today’s population, not a deep natural parameter that tells you about causality. Heritability of smoking went up when smoking became less socially approved, for example.