slight mod making it nicer