Fix for small devices

Hopefully this works good enough