Fix typo / bug