Unlock Settings > Manage Call Behavior

Change the following:

Incoming Phone Call Behavior > Disable on phone ring Disconnected Phone Call Behavior > Re-lock the device right after the call is disconnected Answered Phone Call Behavior > Do not re-lock the device (If I recall correctly, this will allow you to still access the "end call" button)