fixed previous bug fix :P