Fix assembly references

I started getting compiler errors because of missing assembly
references. I'm not sure why I didn't see them before. I added
the new references.