Fixing test and typo