Fixed failing test.
Moar test