Add neglected import