Fix broken tests.

Fixes broken tests at lines 88, 126, and 160.