Bug Fix, id replaced by class in JS