Remove return to the code example