modifying test to ensure correct result