fixing split_step documentation