Fix bug for small device