Fix bugs :grinning: