fixing tests?