Minor fix to test which had some code that wasn't actually doing anything.