Added fix to bug where I forgot to add element selector type in some lines of code