Kraken: improve UT on stay_in