minor changes to pass test