Fixed using wrong unittest assert.