Fix json imports in tests