quick fix bugs

oh dear…