Removed element-wise deprecation warning