Fix customer tests