fixing the code example