Undo test fix, was on wrong branch