Refactor fix

I refactored a method and because of two similarly named methods there
were some errors