Unit test fix for Codeship