More little fixing... oops