updates to test for correctness