Oops, feel really dumb. Fixed previous bug fix