fix tests for now