CDITCK-248 Update assertions and test