quick fix to green tests again