Josh Hazlewood aims to sustain his momentum after a strong Test return against Bangladesh, focusing on upcoming ODIs to sharpen skills for South Africa's Test series in October.