test fix (almost)