Add some testing to mux