BUG FIX missing 2 import in Test.