fix unit tests for trait extensions