one last bug fix