small fix with arrange