Fixed failing tests.

I'm sorry for the small commit.