type fix
minor optimisation