Fixed tests for bug fix.