Fixed some stupid bug