fixed a cheeky bug