Fixing code typo