Updated Koala tests