Fixing mock call on py34