quick fix

forgot some stuff~ oops