Small adjustment to #50