fix tests in models