Fix bugs ya