Update test to test what it says it tests