Fix bugss