Update test to improve readability