Fixed junit issue.