I think this fixes a test