minor bug fiexd