BUG: import statements for mock incorrect