fixed a but hopefully