bug fix hopefully