FIX: Fixed imports in testing core module