Optimisations to element prediction