Head looks to white ball for Test return