Fixed unit tests as they are failing

2 jobs for !39 with implement_custom_operators_api in 49 seconds (queued for 1 second)
merge request