Corrected test assertions