Te pueden despedir si estas de baja por enfermedad?

El 1 de julio empecé a trabajar en una empresa por tres meses por sustitución de vacaciones. Ahora he tenido la mala fortuna de que me he tenido que dar de baja por un pinzamiento en la espalda, el dia 29 de julio, pues no puedo trabajar ya que es de limpieza. Hoy, dia 31 de julio, me han llamado de la empresa y me han preguntado si sabía cuantos días estaría de baja. Al contestarles que hasta el lunes el médico no me diría los días, me han dicho que rescinden mi contrato. El contrato tiene 45 días de periodo de prueba. Quisiera saber si es legal rescindir mi contrato por darme de baja aunque esté dentro de estos 45 días.