Fix type causing import error.