Fix tests to work on 301 to 302 change