Fix copy paste oversight
Missing import