Schema test and final flake
Removed unused import