Fixing select box bug