En este artículo proponemos una modificación del modelo constitutivo de aleación de memoria de forma policristalina originalmente propuesto por Souza. Al introducir una energía de deformación de transformación con dos coeficientes de endurecimiento diferentes, podemos tener en cuenta el efecto de la transformación martensítica de granos orientados desfavorablemente que ocurre después de la meseta principal. Al elegir un coeficiente de endurecimiento secundario adecuado, es posible reproducir el comportamiento correcto de tensión-deformación del material después de la meseta sin necesidad de introducir un módulo de Young mucho más pequeño para la martensita. La modificación propuesta se introduce en el modelo que comprende efectos de deformación permanente. Los resultados del modelo para pruebas de tensión uniaxial se comparan con resultados experimentales mostrando una buena concordancia.