Fix #8 Minor change