minor fix
Updated with NashCoding credits