Fix for test which was not running travis but was accidentally broken.