Changed a small part of code to make stuff nicer for me