Fix tests for last-minute change

All hail Travis CI