Fixed import errors

Fix in the AR unittest