Fixed import error while trying to test without full module path