Fixing tests for mux new responsability