Add mock import