Bug fix

Fixed that annoying bug