Fixing test assert