Updated test import path