Fixed that one bug