fix loading ai maybe probably!?