Fixed a failing test (missing 'use' statement)