Test fix

Fixing up some broken tests from https://github.com/24pullrequests/24pullrequests/pull/761