#12 Fix Bug