Minor change
Improved type clarity