fix tests for now while i refactor