kmselement: Reduce critical regions

Change-Id: Ic8edf9bffb0d799d6e3bb447fbd9cd4fa3c0de09