Fixed missing tests for previous commit.