Try to fix check() issues with predict methods