Unittest fix that was missing from last commit