Fixed test-case after altering the way the wrapper works