Fixes error in #1

Sorry for taking so long!