fix error - thanks @jrrideout