Fix #3 Bug