This line was missing from tests