Fixing bug where it would always choose the first hypothesis