quick fix for the first test.