small cleanup from the last commit