change tests a bit