Minor bug fixed