Importing mock better