Remove mock.patch for no longer existing import.  This should fix the failing test.