Added necessary assembly reference and fixed tests.