Fixed the previous bug fix :)