Fix typo in vae.per_component_kl_divergence_term