Added missing assembly reference