tests: Fix correctness of two tests