Fixing expected output