Minor bug fix in fragment