BUG FIXES