Fix small typo/missing code