Fix bug for warning (temporary solution)

- Warnings were breaking computation. Found a temporary solution.