Minor bug solved