Minor improvements to ecp.c changes

Signed-off-by: David Horstmann <david.horstmann@arm.com>
diff --git a/library/ecp.c b/library/ecp.c
index 48c2403..83df037 100644
--- a/library/ecp.c
+++ b/library/ecp.c
@@ -2395,12 +2395,12 @@
         mbedtls_free( T );
     }
 
-    int should_free_R = 0;
     /* prevent caller from using invalid value */
-    should_free_R = ( ret != 0 );
+    int should_free_R = ( ret != 0 );
 #if defined(MBEDTLS_ECP_RESTARTABLE)
     /* don't free R while in progress in case R == P */
-    should_free_R = should_free_R && ( ret != MBEDTLS_ERR_ECP_IN_PROGRESS );
+    if( ret == MBEDTLS_ERR_ECP_IN_PROGRESS )
+        should_free_R = 0;
 #endif
     if( should_free_R )
         mbedtls_ecp_point_free( R );