small fix layout