Fixed import paths in test directory