fixed some expected output issues