Fixed cosmetic bug in test