Update tests and add missing dependency