Correcting flake error on unused import