Fixed the junit problem for migration task test.
fixed the Junit for module test.