Change test assertion for clarity