Fix assembly reference issue