Bug fix (what the hell did I do here?)