fix tests :-(