Bug fix (yay)