Corrected the return for the stack in split_stack actually