added missing import of assert_