Fix import error introduced in previous commit