Bug fix -- thanks @jorbor