Small bug fixes.

I forgot to include a class and change some method calls in a view.