Fix timeout problem in case of longer calculations