[AC KM] Updated tests for new UI