Fix bug with small device