Fixes failing tests due to previous commit.