Updated di tests