Small interface fix