I've changed the code to use it