Add import to test

Begin