whoops fixing small issue from cleanup PR #1