Fix tests and add missing files.