Fixed test: forgot imports.