Fixing bug that messed up front-end