结论：单位在您休病假两个月后辞退您，这种做法不一定合法，具体要分情况判断，您可以根据不同情况采取相应措施维护自身权益。以下是详细分析：
若您处于医疗期内
- 法律规定：根据《企业职工患病或非因工负伤医疗期规定》，医疗期是指企业职工因患病或非因工负伤停止工作治病休息不得解除劳动合同的时限。医疗期的长度根据本人实际参加工作年限和在本单位工作年限确定，一般为3个月到24个月。您刚上班一个月就生病，若符合相关规定，您应享有相应的医疗期。在医疗期内，单位不得依据《劳动合同法》第四十条（无过失性辞退）和第四十一条（经济性裁员）的规定解除劳动合同。
- 单位行为性质：如果此时单位辞退您，属于违法解除劳动合同。
- 您的应对措施：您可以要求单位继续履行劳动合同，即回到单位继续工作；如果不想继续在该单位工作，也可以要求单位按照经济补偿标准的二倍向您支付赔偿金。经济补偿按劳动者在本单位工作的年限，每满一年支付一个月工资的标准向劳动者支付，六个月以上不满一年的，按一年计算；不满六个月的，向劳动者支付半个月工资的经济补偿。
若医疗期已满
- 法律规定：医疗期满后，如果您不能从事原工作，也不能从事由用人单位另行安排的工作的，用人单位提前三十日以书面形式通知劳动者本人或者额外支付劳动者一个月工资后，可以解除劳动合同。但用人单位需要承担举证责任，证明您不能从事原工作以及另行安排的工作。
- 单位行为性质：
- 如果单位没有提前三十日以书面形式通知您，也没有额外支付一个月工资，就直接辞退您，或者无法证明您不能从事原工作和另行安排的工作，那么单位的解除行为属于违法解除，您有权要求单位继续履行合同或者支付赔偿金。
- 如果单位履行了相应的程序，并且有充分证据证明您不能从事原工作和另行安排的工作，那么单位的解除行为是合法的。但单位仍需按照规定向您支付经济补偿。
- 您的应对措施：
- 若单位违法解除，您可以先与单位协商，要求其纠正违法行为；协商不成，可以向劳动争议调解组织申请调解，或者直接向劳动争议仲裁委员会申请仲裁，对仲裁裁决不服的，还可以向人民法院提起诉讼。
- 若单位合法解除，您要确认单位是否按照规定支付了经济补偿，如果没有支付，您可以要求单位支付。