Minimal change to pass second test