Fixed so it works on the pi again