Remove unused import mock