Fixed Minor bug