made test pass with new changes